package com.yhw.srb.oss.service.impl;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.CannedAccessControlList;
import com.yhw.srb.oss.service.FileService;
import com.yhw.srb.oss.util.OssProperties;
import org.apache.commons.lang3.StringUtils;
import org.joda.time.DateTime;
import org.springframework.stereotype.Service;

import java.io.InputStream;
import java.util.UUID;

@Service
public class FileServiceImpl implements FileService {

    /**
     * 文件上传至阿里云    （之前用过 七牛云 ）
     */
    @Override
    public String upload(InputStream inputStream, String module, String fileName) {
        //  inputStream 上传文件的输入流     module 目录     fileName 文件名

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(OssProperties.ENDPOINT, OssProperties.KEY_ID, OssProperties.KEY_SECRET);
        //判断bucket（空间）是否存在：如果不存在则创建，如果存在则获取
        if(!ossClient.doesBucketExist(OssProperties.BUCKET_NAME)){
            //创建bucket
            ossClient.createBucket(OssProperties.BUCKET_NAME);
            //设置oss实例的访问权限：公共读   (公共读 其他人只能读不能写  即只能下载 不能上传)
            ossClient.setBucketAcl(OssProperties.BUCKET_NAME,CannedAccessControlList.PublicRead);
        }

        // 填写文件完整路径，完整路径中不能包含Bucket名称，例如exampledir/exampleobject.txt
        // 比如 https://yhw-srb-file.oss-cn-hangzhou.aliyuncs.com/avatar/1.jpg  只要 avatar/1.jpg
        // 构建日期路径：avatar/2019/02/26/文件名

        String folder = new DateTime().toString("yyyy/MM/dd");

        // 不使用原始文件名  因为可能会乱码 或者 冲突   比如： 上传 a.png(蔡徐坤) 接着 上传 a.png(刘德华)  冲突
        // 唯一文件名  uuid.     StringUtils.substringAfterLast(filename, ".")  取得文件名后缀  这个类在 commons.lang3 包下
        fileName = UUID.randomUUID().toString() + "." + StringUtils.substringAfterLast(fileName,".");

        String objectName = module + "/" + folder + "/" + fileName;

        // 文件上传至阿里云
        ossClient.putObject(OssProperties.BUCKET_NAME, objectName, inputStream);

        // 关闭OSSClient。
        ossClient.shutdown();

        // 上传成功 要把文件的 url 返回给前端
        // 比如 https://yhw-srb-file.oss-cn-hangzhou.aliyuncs.com/avatar/1.jpg
        return "https://" + OssProperties.BUCKET_NAME + "." + StringUtils.substringAfterLast(OssProperties.ENDPOINT,"/")+ "/" + objectName;

        //  https://yhw-srb-file.oss-cn-hangzhou.aliyuncs.com/test/2022/09/06/6bf3bcc6-6b7c-4ef1-aa77-5adfc689ab0c.jpg
    }

    @Override
    public void remove(String url) {

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(OssProperties.ENDPOINT, OssProperties.KEY_ID, OssProperties.KEY_SECRET);

        // https://yhw-srb-file.oss-cn-hangzhou.aliyuncs.com/
        String s = "https://" + OssProperties.BUCKET_NAME + "." + StringUtils.substringAfterLast(OssProperties.ENDPOINT,"/")+ "/";

        // 填写文件完整路径。完整路径中不能包含Bucket名称，例如exampledir/exampleobject.txt
        String objectName = url.substring(s.length());

        // 删除文件或目录。如果要删除目录，目录必须为空。
        ossClient.deleteObject(OssProperties.BUCKET_NAME, objectName);

        // 关闭OSSClient。
        ossClient.shutdown();
    }

}
